package com.haidechizi.leetcode;


/**
 * 还记得童话《卖火柴的小女孩》吗？现在，你知道小女孩有多少根火柴，请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴，可以把火柴连接起来，并且每根火柴都要用到。
 * <p>
 * 输入为小女孩拥有火柴的数目，每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
 * <p>
 * 示例 1:
 * <p>
 * 输入: [1,1,2,2,2]
 * 输出: true
 * <p>
 * 解释: 能拼成一个边长为2的正方形，每边两根火柴。
 * 示例 2:
 * <p>
 * 输入: [3,3,3,3,4]
 * 输出: false
 * <p>
 * 解释: 不能用所有火柴拼成一个正方形。
 * 注意:
 * <p>
 * 给定的火柴长度和在 0 到 10^9之间。
 * 火柴数组的长度不超过15。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/matchsticks-to-square
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class _473_火柴拼正方形 {

    public boolean makesquare(int[] nums) {
        if (nums == null || nums.length < 4) {
            return false;
        }
        int sum = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
        }
        if (sum % 4 != 0) {
            return false;
        }
        int[] bucket = new int[4];
        return generate(0, nums, bucket, sum / 4);
    }

    public boolean generate(int i, int[] nums, int[] bucket, int target) {
        if (i >= nums.length) {
            return bucket[0] == target && bucket[1] == target && bucket[2] == target && bucket[3] == target;
        }
        for (int j = 0; j < 4; j++) {
            if (bucket[j] + nums[i] > target) {
                continue;
            }
            bucket[j] += nums[i];
            if (generate(i + 1, nums, bucket, target)) {
                return true;
            }
            bucket[j] -= nums[i];

        }
        return false;
    }

    public static void main(String[] args) {
        System.out.println(new _473_火柴拼正方形().makesquare(new int[]
                {5, 5, 5, 5, 4, 4, 4, 4, 3, 3, 3, 3}));
    }


}
